transform =transforms.Compose([transforms.Resize((224,224)),
                                    transforms.ToTensor()])
folder_dataset = dset.ImageFolder(root=training_dir)
siamese_dataset = SiameseNetworkDataset(imageFolderDataset=folder_dataset,
                                                transform=transform,
                                                should_invert=False)
train_dataloader = DataLoader(siamese_dataset,
                                    shuffle=True,
                                    num_workers=0,
                                    batch_size=train_batch_size) 
